package com.company.project.service;

import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
import com.company.project.entity.ScheduledTaskEntity;
import com.company.project.entity.ScheduledTaskLogEntity;
import com.company.project.service.ScheduledTaskLogService;
import lombok.extern.slf4j.Slf4j;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.scheduling.annotation.Scheduled;
import org.springframework.stereotype.Service;

import java.util.Date;
import java.util.List;
import java.util.Map;
import java.util.concurrent.ConcurrentHashMap;

/**
 * 定时任务执行器服务
 * 根据每个任务的执行间隔来定期输出调用目标字符串
 */
@Service
@Slf4j
public class ScheduledTaskExecutorService {

    @Autowired
    private ScheduledTaskService scheduledTaskService;

    @Autowired
    private ScheduledTaskLogService scheduledTaskLogService;

    // 存储每个任务的最后执行时间
    private final Map<Integer, Long> taskLastExecuteTime = new ConcurrentHashMap<>();

    /**
     * 每1秒执行一次，检查并执行启用的定时任务
     * 这样可以支持1秒间隔的快速任务
     */
    @Scheduled(fixedRate = 1000) // 每1秒执行一次
    public void executeScheduledTasks() {
        try {
            // 获取所有启用的定时任务
            List<ScheduledTaskEntity> enabledTasks = scheduledTaskService.list(
                new QueryWrapper<ScheduledTaskEntity>()
                    .eq("status", 1)
                    .eq("deleted", 0)
            );

            for (ScheduledTaskEntity task : enabledTasks) {
                try {
                    executeTask(task);
                } catch (Exception e) {
                    log.error("执行定时任务失败，任务ID: {}, 任务名称: {}, 错误: {}", 
                        task.getId(), task.getTaskName(), e.getMessage());
                }
            }
        } catch (Exception e) {
            log.error("定时任务执行器异常: {}", e.getMessage());
        }
    }

    /**
     * 执行具体的定时任务
     * 根据任务的执行间隔来决定是否执行
     */
    private void executeTask(ScheduledTaskEntity task) {
        long currentTime = System.currentTimeMillis();
        long lastExecuteTime = taskLastExecuteTime.getOrDefault(task.getId(), 0L);
        long intervalMillis = task.getIntervalSeconds() * 1000L;

        // 检查是否到了执行时间
        if (currentTime - lastExecuteTime >= intervalMillis) {
            long startTime = System.currentTimeMillis();
            ScheduledTaskLogEntity logEntity = new ScheduledTaskLogEntity();
            logEntity.setTaskId(task.getId());
            logEntity.setTaskName(task.getTaskName());
            logEntity.setTargetString(task.getTargetString());
            logEntity.setExecuteTime(new Date());
            logEntity.setCreateTime(new Date());
            
            try {
                // 在控制台输出调用目标字符串
                System.out.println("=== 定时任务执行 ===");
                System.out.println("任务ID: " + task.getId());
                System.out.println("任务名称: " + task.getTaskName());
                System.out.println("执行间隔: " + task.getIntervalSeconds() + "秒");
                System.out.println("调用目标字符串: " + task.getTargetString());
                System.out.println("执行时间: " + new java.util.Date());
                System.out.println("==================");
                
                log.info("开始执行定时任务: {}, 目标字符串: {}", task.getTaskName(), task.getTargetString());
                
                // 处理目标字符串
                processTargetString(task.getTargetString());
                
                // 记录成功日志
                long endTime = System.currentTimeMillis();
                logEntity.setExecuteStatus(1); // 成功
                logEntity.setExecuteMessage("执行成功");
                logEntity.setExecuteDuration(endTime - startTime);
                
                log.info("定时任务执行完成: {}", task.getTaskName());
            } catch (Exception e) {
                // 记录失败日志
                long endTime = System.currentTimeMillis();
                logEntity.setExecuteStatus(0); // 失败
                logEntity.setExecuteMessage("执行失败: " + e.getMessage());
                logEntity.setExecuteDuration(endTime - startTime);
                
                log.error("定时任务执行失败: {}, 错误: {}", task.getTaskName(), e.getMessage());
            } finally {
                // 保存执行日志
                try {
                    scheduledTaskLogService.save(logEntity);
                } catch (Exception e) {
                    log.error("保存定时任务执行日志失败: {}", e.getMessage());
                }
            }
            
            // 更新最后执行时间
            taskLastExecuteTime.put(task.getId(), currentTime);
        }
    }

    /**
     * 处理目标字符串，根据JSON内容执行相应的业务逻辑
     */
    private void processTargetString(String targetString) {
        try {
            // 这里可以根据targetString的JSON内容来执行不同的业务逻辑
            // 例如：
            // {"type": "http", "url": "http://example.com/api", "method": "POST"}
            // {"type": "database", "sql": "UPDATE table SET field = value"}
            // {"type": "email", "to": "user@example.com", "subject": "定时通知"}
            
            log.info("处理目标字符串: {}", targetString);
            
            // 示例：简单的日志记录
            // 实际项目中可以根据JSON内容解析并执行相应的业务逻辑
            
        } catch (Exception e) {
            log.error("处理目标字符串失败: {}, 错误: {}", targetString, e.getMessage());
        }
    }

    /**
     * 获取任务执行统计信息
     */
    public Map<String, Object> getTaskExecutionStats() {
        Map<String, Object> stats = new ConcurrentHashMap<>();
        stats.put("totalTasks", taskLastExecuteTime.size());
        stats.put("lastExecuteTimes", taskLastExecuteTime);
        return stats;
    }
}
